package org.jeecg.common.system.base.controller;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.IService;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.beanutils.PropertyUtils;
import org.apache.shiro.SecurityUtils;
import org.jeecg.common.api.vo.Result;
import org.jeecg.common.system.query.QueryGenerator;
import org.jeecg.common.system.vo.LoginUser;
import org.jeecg.common.util.oConvertUtils;
import org.jeecgframework.poi.excel.ExcelImportUtil;
import org.jeecgframework.poi.excel.def.NormalExcelConstants;
import org.jeecgframework.poi.excel.entity.ExportParams;
import org.jeecgframework.poi.excel.entity.ImportParams;
import org.jeecgframework.poi.excel.view.JeecgEntityExcelView;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.MultipartHttpServletRequest;
import org.springframework.web.servlet.ModelAndView;

/** @Description: Controller基类 @Author: dangzhenghui@163.com @Date: 2019-4-21 8:13 @Version: 1.0 */
@Slf4j
public class JeecgController<T, S extends IService<T>> {
  @Autowired S service;

  @Value("${jeecg.path.upload}")
  private String upLoadPath;
  /**
   * 导出excel
   *
   * @param request
   */
  protected ModelAndView exportXls(
      HttpServletRequest request, T object, Class<T> clazz, String title) {
    // Step.1 组装查询条件
    QueryWrapper<T> queryWrapper =
        QueryGenerator.initQueryWrapper(object, request.getParameterMap());
    LoginUser sysUser = (LoginUser) SecurityUtils.getSubject().getPrincipal();

    // Step.2 获取导出数据
    List<T> pageList = service.list(queryWrapper);
    List<T> exportList = null;

    // 过滤选中数据
    String selections = request.getParameter("selections");
    if (oConvertUtils.isNotEmpty(selections)) {
      List<String> selectionList = Arrays.asList(selections.split(","));
      exportList =
          pageList.stream()
              .filter(item -> selectionList.contains(getId(item)))
              .collect(Collectors.toList());
    } else {
      exportList = pageList;
    }

    // Step.3 AutoPoi 导出Excel
    ModelAndView mv = new ModelAndView(new JeecgEntityExcelView());
    mv.addObject(NormalExcelConstants.FILE_NAME, title); // 此处设置的filename无效 ,前端会重更新设置一下
    mv.addObject(NormalExcelConstants.CLASS, clazz);
    // update-begin--Author:liusq  Date:20210126 for：图片导出报错，ImageBasePath未设置--------------------
    ExportParams exportParams =
        new ExportParams(title + "报表", "导出人:" + sysUser.getRealname(), title);
    exportParams.setImageBasePath(upLoadPath);
    // update-end--Author:liusq  Date:20210126 for：图片导出报错，ImageBasePath未设置----------------------
    mv.addObject(NormalExcelConstants.PARAMS, exportParams);
    mv.addObject(NormalExcelConstants.DATA_LIST, exportList);
    return mv;
  }

  /**
   * 根据权限导出excel，传入导出字段参数
   *
   * @param request
   */
  protected ModelAndView exportXls(
      HttpServletRequest request, T object, Class<T> clazz, String title, String exportFields) {
    ModelAndView mv = this.exportXls(request, object, clazz, title);
    mv.addObject(NormalExcelConstants.EXPORT_FIELDS, exportFields);
    return mv;
  }

  /**
   * 获取对象ID
   *
   * @return
   */
  private String getId(T item) {
    try {
      return PropertyUtils.getProperty(item, "id").toString();
    } catch (Exception e) {
      e.printStackTrace();
      return null;
    }
  }

  /**
   * 通过excel导入数据
   *
   * @param request
   * @param response
   * @return
   */
  protected Result<?> importExcel(
      HttpServletRequest request, HttpServletResponse response, Class<T> clazz) {
    MultipartHttpServletRequest multipartRequest = (MultipartHttpServletRequest) request;
    Map<String, MultipartFile> fileMap = multipartRequest.getFileMap();
    for (Map.Entry<String, MultipartFile> entity : fileMap.entrySet()) {
      MultipartFile file = entity.getValue(); // 获取上传文件对象
      ImportParams params = new ImportParams();
      params.setTitleRows(2);
      params.setHeadRows(1);
      params.setNeedSave(true);
      try {
        List<T> list = ExcelImportUtil.importExcel(file.getInputStream(), clazz, params);
        // update-begin-author:taoyan date:20190528 for:批量插入数据
        long start = System.currentTimeMillis();
        service.saveBatch(list);
        // 400条 saveBatch消耗时间1592毫秒  循环插入消耗时间1947毫秒
        // 1200条  saveBatch消耗时间3687毫秒 循环插入消耗时间5212毫秒
        log.info("消耗时间" + (System.currentTimeMillis() - start) + "毫秒");
        // update-end-author:taoyan date:20190528 for:批量插入数据
        return Result.ok("文件导入成功！数据行数：" + list.size());
      } catch (Exception e) {
        log.error(e.getMessage(), e);
        return Result.error("文件导入失败:" + e.getMessage());
      } finally {
        try {
          file.getInputStream().close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }
    return Result.error("文件导入失败！");
  }
}
